import csv
import os

# 假设存在一个名为 xiaofei.csv 的文件，其中包含消费数据
if not os.path.exists("xiaofei.csv"):
    print("文件xiaofei.csv 未找到，请确保文件存在。")
    print("请创建一个包含以下表头的文件：省份,姓名,年龄,性别,消费金额")
    exit()

try:
    with open("xiaofei.csv", "r", encoding="gbk") as f:
        reader = csv.reader(f)
        header = next(reader)  # 读取表头
        data = list(reader)  # 读取数据
except FileNotFoundError:
    print("xiaofei.csv 未找到，请确保文件存在。")
    exit()

# 获取用户输入的省份和项目
province = input("请输入要查询的省份：")
items = input("请输入要查询的项目，用空格分隔：").split()

# 筛选数据
results = []
for row in data:
    if row[0] == province:  # 假设省份在第一列
        result = {}
        for i, item in enumerate(items):
            try:
                index = header.index(item)  # 获取项目对应的列索引
                result[item] = row[index]  # 将项目和对应的值添加到结果字典中
            except ValueError:
                print(f"项目 {item} 不存在。")
                continue
        if result:
            results.append(result)

# 输出结果
if results:
    print("查询结果：")
    for result in results:
        for key, value in result.items():
            print(f"{key}: {value}")
else:
    print("没有找到符合条件的数据。")
